/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file           : main.c
  * @brief          : Main program body
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2025 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"
#include "usart.h"
#include "gpio.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */
#include "stdio.h"
#include "flash.h"
#include "string.h"
/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */

/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */
uint64_t Upgrade_Flag;//????
uint64_t NotUpgrade = {0xFFFFFFFFFFFFFFFF};
uint64_t Upgrade =    {0xAAAAAAAAAAAAAAAA};

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/

/* USER CODE BEGIN PV */

/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);
/* USER CODE BEGIN PFP */

// ******************************** �����ʼ���� ***********************************//
#define  FLASH_APP_ADDR  (0x08020000)//APP1

typedef  void (*pFunction)(void);
pFunction Jump_To_Application;
void Jump_to_APP(uint32_t app_addr) {
    uint32_t sp = *(volatile uint32_t*)app_addr;
    uint32_t pc = *(volatile uint32_t*)(app_addr + 4);
    printf("jump to app: %#x\r\n", FLASH_APP_ADDR);
    // 校验SP是否指向RAM（0x20000000~0x2003FFFF）
    if ((sp & 0xFFF00000) == 0x20000000) {  // F407的RAM为256KB（0x20000000~0x2003FFFF）
        __disable_irq();
        __set_MSP(sp);
			printf("jump %#x success \r\n",FLASH_APP_ADDR);
        ((void (*)())pc)();  // 直接跳转
    } else {
        printf("Jump Error: SP=0x%08X, PC=0x%08X\r\n", sp, pc);
    }
}


void print_boot_message(void) {
    printf("---------- Enter BootLoader ----------\r\n");
    printf("======== Flash Partition Table =======\r\n");
    printf("| Name     | Start Addr  |   Size    |\r\n");
    printf("--------------------------------------\r\n");
    printf("| Boot     | 0x08000000  |  128 KB   |\r\n");
    printf("| App1     | 0x08020000  |  128 KB   |\r\n");
    printf("| App2     | 0x08040000  |  128 KB   |\r\n");
    printf("| Flag     | 0x080E0000  |  128 KB   |\r\n");
    printf("======================================\r\n");
}

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */
/************************************************************
*��������Copy_APP2_to_APP1
*��  �ܣ�APP2������APP1��ַ��
*��  ������
*����ֵ����
*************************************************************/

uint64_t u64_Code_Buff[256];//2k
uint16_t Num = 0;
void Copy_APP2_to_APP1(void) {
    uint32_t APP1_Addr = FLASH_APP1_START_ADDR;
    uint32_t APP2_Addr = FLASH_APP2_START_ADDR;

    // 擦除APP1所在扇区（扇区5）
    MyErase_Sectors(FLASH_APP1_START_ADDR, APP1_SECTOR_NUM);
    printf("Erased App1 Sector %d (128KB)\r\n", APP1_SECTOR_NUM);

    // 复制APP2到APP1（每次写入2KB，共64次=128KB）
    for(int i = 0; i < 64; i++) {  // 128KB / 2KB = 64
        MyRead64Flash(APP2_Addr, u64_Code_Buff, 256);
        MyWrite64Flash(APP1_Addr, u64_Code_Buff, 256);

        APP1_Addr += 0x800;  // 2KB偏移
        APP2_Addr += 0x800;

        // 打印进度
        printf("Copied block %d: App2(0x%08X) -> App1(0x%08X)\r\n", 
              i, APP2_Addr - 0x800, APP1_Addr - 0x800);
        memset(u64_Code_Buff, 0, sizeof(u64_Code_Buff));
    }

    // 擦除APP2所在扇区（扇区6）
    MyErase_Sectors(FLASH_APP2_START_ADDR, APP2_SECTOR_NUM);
    printf("Erased App2 Sector %d (128KB)\r\n", APP2_SECTOR_NUM);

    // 清除升级标志
    InFlashWrite(UPGRADE_FLAG_ADDR, NotUpgrade);
}


/* USER CODE END 0 */

/**
  * @brief  The application entry point.
  * @retval int
  */
int main(void)
{

  /* USER CODE BEGIN 1 */

  /* USER CODE END 1 */

  /* MCU Configuration--------------------------------------------------------*/

  /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
  HAL_Init();

  /* USER CODE BEGIN Init */

  /* USER CODE END Init */

  /* Configure the system clock */
  SystemClock_Config();

  /* USER CODE BEGIN SysInit */

  /* USER CODE END SysInit */

  /* Initialize all configured peripherals */
  MX_GPIO_Init();
  MX_USART1_UART_Init();
  /* USER CODE BEGIN 2 */
	// 在main函数初始化后添加：
if (FLASH_R(UPGRADE_FLAG_ADDR) != NotUpgrade && FLASH_R(UPGRADE_FLAG_ADDR) != Upgrade) {
    printf("Initializing Upgrade Flag...\r\n");
    InFlashWrite(UPGRADE_FLAG_ADDR, NotUpgrade); // 初始化为未升级状态
}

  printf("bootloader running\r\n");
  print_boot_message();
	
	// 读取升级标志
    Upgrade_Flag = FLASH_R(UPGRADE_FLAG_ADDR);
    printf("Upgrade Flag: 0x%016llX\r\n", Upgrade_Flag);

    if (Upgrade_Flag == NotUpgrade) {
        printf("Booting App1...\r\n");
        Jump_to_APP(FLASH_APP_ADDR);
    } else if (Upgrade_Flag == Upgrade) {
        printf("Upgrading App2 to App1...\r\n");
        Copy_APP2_to_APP1();
        printf("Upgrade Complete. Booting App1...\r\n");
        Jump_to_APP(FLASH_APP_ADDR);
    } else {
        printf("Invalid Upgrade Flag! Halting.\r\n");
        while(1);
    }

  /* USER CODE END 2 */

  /* Infinite loop */
  /* USER CODE BEGIN WHILE */
  while (1)
  {
    /* USER CODE END WHILE */

    /* USER CODE BEGIN 3 */
  }
  /* USER CODE END 3 */
}

/**
  * @brief System Clock Configuration
  * @retval None
  */
void SystemClock_Config(void)
{
  RCC_OscInitTypeDef RCC_OscInitStruct = {0};
  RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

  /** Configure the main internal regulator output voltage
  */
  __HAL_RCC_PWR_CLK_ENABLE();
  __HAL_PWR_VOLTAGESCALING_CONFIG(PWR_REGULATOR_VOLTAGE_SCALE1);

  /** Initializes the RCC Oscillators according to the specified parameters
  * in the RCC_OscInitTypeDef structure.
  */
  RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSE;
  RCC_OscInitStruct.HSEState = RCC_HSE_ON;
  RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
  RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSE;
  RCC_OscInitStruct.PLL.PLLM = 25;
  RCC_OscInitStruct.PLL.PLLN = 336;
  RCC_OscInitStruct.PLL.PLLP = RCC_PLLP_DIV2;
  RCC_OscInitStruct.PLL.PLLQ = 4;
  if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
  {
    Error_Handler();
  }

  /** Initializes the CPU, AHB and APB buses clocks
  */
  RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK|RCC_CLOCKTYPE_SYSCLK
                              |RCC_CLOCKTYPE_PCLK1|RCC_CLOCKTYPE_PCLK2;
  RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_PLLCLK;
  RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
  RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV4;
  RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV2;

  if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_5) != HAL_OK)
  {
    Error_Handler();
  }
}

/* USER CODE BEGIN 4 */

/* USER CODE END 4 */

/**
  * @brief  This function is executed in case of error occurrence.
  * @retval None
  */
void Error_Handler(void)
{
  /* USER CODE BEGIN Error_Handler_Debug */
  /* User can add his own implementation to report the HAL error return state */
  __disable_irq();
  while (1)
  {
  }
  /* USER CODE END Error_Handler_Debug */
}

#ifdef  USE_FULL_ASSERT
/**
  * @brief  Reports the name of the source file and the source line number
  *         where the assert_param error has occurred.
  * @param  file: pointer to the source file name
  * @param  line: assert_param error line source number
  * @retval None
  */
void assert_failed(uint8_t *file, uint32_t line)
{
  /* USER CODE BEGIN 6 */
  /* User can add his own implementation to report the file name and line number,
     ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
  /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */
